from ocr.cert.card import Card
from pathlib import Path
import unittest

class TestCard(unittest.TestCase):
    def test_get_sfz(self):
        current_file_path = Path(__file__).resolve()
        file_path = f"{current_file_path.parent}/身份证样证.png"
        card = Card()
        actual = card.get_sfz(file_path)
        expected = {'status': 0,
                    'lists': {'name': {'label': '姓名', 'value': '李久熙'}, 'sex': {'label': '性别', 'value': '女'},
                              'nation': {'label': '民族', 'value': '汉'},
                              'birth': {'label': '出生', 'value': '1996年11月24日'},
                              'address': {'label': '住址',
                                          'value': '北京市海淀区双榆树东里住址北京市海淀区双榆树东里99区2号楼302室'},
                              'code': {'label': '公民身份号码', 'value': '110108199611240188'}}}
        assert expected == actual
